package com.example.demo.controller;

import com.auth0.jwt.JWT;
import com.example.demo.entity.User;
import com.example.demo.entity.VO.LoginUser;
import com.example.demo.service.UserService;
import com.example.demo.utils.JWTUtils;
import com.example.demo.utils.PassToken;
import com.example.demo.utils.ServerResponse;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;
import java.util.HashMap;
import java.util.Map;

@RestController
public class UserController {
    @Autowired
    UserService userService;

    @Autowired
    RedisTemplate<String,Object> redisTemplate;

    @PostMapping("/login")
    public ServerResponse<String> login(@RequestBody LoginUser loginUser){
        String storeCode = (String) redisTemplate.opsForValue().get("captcha_codes:"+loginUser.getUuid());
        System.out.println(storeCode+loginUser.toString());
        if(loginUser.getCode().equals(storeCode)) {

            User u=userService.getUser(loginUser);
            if(u!=null){
            Map<String, String> map = new HashMap<>();
            map.put("userid", u.getUserId()+"");
            map.put("username", u.getUserName());
            String token = JWTUtils.getToken(map);
            redisTemplate.delete(storeCode);
                return ServerResponse.createBySuccess(200, token, "登录成功");
            }
            return ServerResponse.createBySuccess(500, "", "账号或密码错误");
        }
        else{
            return ServerResponse.createBySuccess(500,"","验证码错误");
        }
    }

    @GetMapping("getInfo")
    public ServerResponse<User> getInfo(HttpServletRequest request) {
        String token = request.getHeader("Authorization");
        System.out.println(token+"   ");
        String id = JWT.decode(token.substring(token.lastIndexOf(" ")+1)).getClaim("userid").asString();
        User user = userService.getUserById(Integer.parseInt(id));
        return ServerResponse.createBySuccess(200,user,"查询成功");
    }

}
